import requests
import unittest

def get_header():
    url = "http://ihrm-test.itheima.net/api/sys/login"
    data = {"mobile": "13800000002", "password": "123456"}
    resp = requests.post(url=url, json=data)
    print(resp.json())
    # 从 响应体中，获取 data的值
    token = resp.json().get("data")

    header = {"Content-Type": "application/json",
              "Authorization": "Bearer " + token}
    return header


# 在 scripts/ 的测试脚本文件中，添加 setUpClass 方法，调用 get_header 函数。 将返回值 保存到 类属性上
from common.get_header import get_header


class TestEmpAdd(unittest.TestCase):
    # 类属性
    header = None

    @classmethod
    def setUpClass(cls) -> None:
        cls.header = get_header()
